To Copy or Not to Copy: Controlling Speculative Decoding via Intrinsic Model Signals
何时复制:用模型内在信号控制投机解码
🧭 LLM 推理加速 · 投机解码
SwitchSD 用轻量探针识别 LLM 的复制意图,动态切换神经草稿与上下文复制,吞吐最高提升 15%
📌 工作参考价值(4 点)
- 在部署 LLM 推理服务(如推荐文案生成、Agent 工具输出)时,可借鉴 SwitchSD 的 copy-intent 探针:对目标模型中间层训练轻量分类器,判断当前 token 是否应从上下文复制,避免 Prompt Lookup Decoding 的误触发。
- 对于大量 copy 场景(商品描述扩写、基于历史行为生成推荐理由、模板填充),混合策略明显优于单一神经草稿;用 hidden-state probe 代替 n-gram 重叠启发式,可减少意外重复。
- 工程上可将 EAGLE/PLD 作为两个 draft 后端,通过探针分数做低成本路由,无需重新训练主模型;如果业务里 copy 密度高,预期可获得 10%+ 吞吐提升。
- 可复用的结论:copy 不是一个表面 n-gram 模式,而是 LLM 内部可检测的控制信号;把复制判断从规则升级为模型感知,能直接提升投机解码效率。